import os, sys
api_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "api")
sys.path.append(api_path)
import api_llm
import api_prompt_engineering
import yaml

class Agent():
    def __init__(self, **kwargs):
        self.TEMPLATE_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "template")
        self.headers = {
            'Content-Type': 'application/json',
            'Accept': 'application/json',
        }
        self.model = kwargs.get("model", "glm-4-flash")
        
        self.PE = api_prompt_engineering.PromptEngineering()
        self.LLM = api_llm.LLM(debug=kwargs.get("DEBUG_LLM", False), **kwargs)
        
        self.DEBUG = kwargs.get("DEBUG", False)
        
        

if __name__=="__main__":
    agent = Agent()
    print(agent.LLM.chat("你好", model="glm-4-airx"))